package com.mavenq.fly.test.sort;

import java.util.Arrays;

public class QuickSort {

    // 快速排序算法
    public static void quickSort(int[] arr, int low, int high) {
        if (low < high) {
            // 找到基准元素的正确位置
            int pi = partition(arr, low, high);

            // 对基准元素左侧的子数组进行排序
            quickSort(arr, low, pi - 1);

            // 对基准元素右侧的子数组进行排序
            quickSort(arr, pi + 1, high);
        }
    }

    // 分区函数，返回基准元素的索引
    private static int partition(int[] arr, int low, int high) {
        // 选择最后一个元素作为基准元素
        int pivot = arr[high];
        int i = (low - 1); // 最小元素的索引

        for (int j = low; j < high; j++) {
            // 如果当前元素小于或等于基准元素
            if (arr[j] <= pivot) {
                i++;
                // 交换元素
                swap(arr, i, j);
            }
        }
        // 将基准元素放到正确的位置
        swap(arr, i + 1, high);
        return i + 1;
    }

    // 交换数组中两个元素的位置
    private static void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }

    // 测试快速排序的功能
    public static void main(String[] args) {
        int[] arr = {12, 4, 6, 7, 2, 10, 8};
        System.out.println("原始数组: " + Arrays.toString(arr));

        quickSort(arr, 0, arr.length - 1);

        System.out.println("排序后的数组: " + Arrays.toString(arr));
    }
}
